/*
 * Copyright (C) 2005-2011 MaNGOS <http://www.getmangos.com/>
 *
 * Copyright (C) 2008-2011 Trinity <http://www.trinitycore.org/>
 *
 * Copyright (C) 2010-2011 ArkCORE <http://www.arkania.net/>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#ifndef ARKCORE_FORMULAS_H
#define ARKCORE_FORMULAS_H

#include "World.h"
#include "SharedDefines.h"
#include "ScriptMgr.h"

namespace Trinity {
namespace Honor {
inline float hk_honor_at_level_f(uint8 level, float multiplier = 1.0f) {
	float honor = multiplier * level * 1.55f;
	sScriptMgr->OnHonorCalculation(honor, level, multiplier);
	return honor;
}

inline uint32 hk_honor_at_level(uint8 level, float multiplier = 1.0f) {
	return uint32(ceil(hk_honor_at_level_f(level, multiplier)));
}
}
namespace XP {
inline uint8 GetGrayLevel(uint8 pl_level) {
	uint8 level;

	if (pl_level <= 5)
		level = 0;
	else if (pl_level <= 39)
		level = pl_level - 5 - pl_level / 10;
	else if (pl_level <= 59)
		level = pl_level - 1 - pl_level / 5;
	else
		level = pl_level - 9;

	sScriptMgr->OnGrayLevelCalculation(level, pl_level);
	return level;
}

inline XPColorChar GetColorCode(uint8 pl_level, uint8 mob_level) {
	XPColorChar color;

	if (mob_level >= pl_level + 5)
		color = XP_RED;
	else if (mob_level >= pl_level + 3)
		color = XP_ORANGE;
	else if (mob_level >= pl_level - 2)
		color = XP_YELLOW;
	else if (mob_level > GetGrayLevel(pl_level))
		color = XP_GREEN;
	else
		color = XP_GRAY;

	sScriptMgr->OnColorCodeCalculation(color, pl_level, mob_level);
	return color;
}

inline uint8 GetZeroDifference(uint8 pl_level) {
	uint8 diff;

	if (pl_level < 8)
		diff = 5;
	else if (pl_level < 10)
		diff = 6;
	else if (pl_level < 12)
		diff = 7;
	else if (pl_level < 16)
		diff = 8;
	else if (pl_level < 20)
		diff = 9;
	else if (pl_level < 30)
		diff = 11;
	else if (pl_level < 40)
		diff = 12;
	else if (pl_level < 45)
		diff = 13;
	else if (pl_level < 50)
		diff = 14;
	else if (pl_level < 55)
		diff = 15;
	else if (pl_level < 60)
		diff = 16;
	else
		diff = 17;

	sScriptMgr->OnZeroDifferenceCalculation(diff, pl_level);
	return diff;
}

inline uint32 BaseGain(uint8 pl_level, uint8 mob_level, ContentLevels content) {
	uint32 baseGain;
	uint32 nBaseExp;

	switch (content) {
	case CONTENT_1_60:
		nBaseExp = 45;
		break;
	case CONTENT_61_70:
		nBaseExp = 235;
		break;
	case CONTENT_71_80:
		nBaseExp = 580;
		break;
	case CONTENT_81_85:
		nBaseExp = 1878;
		break;
	default:
		sLog->outError("BaseGain: Unsupported content level %u", content);
		nBaseExp = 45;
		break;
	}

	if (mob_level >= pl_level) {
		uint8 nLevelDiff = mob_level - pl_level;
		if (nLevelDiff > 4)
			nLevelDiff = 4;

		baseGain = ((pl_level * 5 + nBaseExp) * (20 + nLevelDiff) / 10 + 1) / 2;
	} else {
		uint8 gray_level = GetGrayLevel(pl_level);
		if (mob_level > gray_level) {
			uint8 ZD = GetZeroDifference(pl_level);
			baseGain = (pl_level * 5 + nBaseExp) * (ZD + mob_level - pl_level)
					/ ZD;
		} else
			baseGain = 0;
	}

	sScriptMgr->OnBaseGainCalculation(baseGain, pl_level, mob_level, content);
	return baseGain;
}

inline uint32 Gain(Player *pl, Unit *u) {
	uint32 gain;

	if (u->GetTypeId() == TYPEID_UNIT
			&& (((Creature*) u)->isTotem() || ((Creature*) u)->isPet()
					|| (((Creature*) u)->GetCreatureInfo()->flags_extra
							& CREATURE_FLAG_EXTRA_NO_XP_AT_KILL)
					|| ((Creature*) u)->GetCreatureInfo()->type
							== CREATURE_TYPE_CRITTER))
		gain = 0;
	else {
		gain = BaseGain(pl->getLevel(), u->getLevel(),
				GetContentLevelsForMapAndZone(u->GetMapId(), u->GetZoneId()));

		if (gain != 0 && u->GetTypeId() == TYPEID_UNIT
				&& ((Creature*) u)->isElite()) {
			// Elites in instances have a 2.75x XP bonus instead of the regular 2x world bonus.
			if (u->GetMap() && u->GetMap()->IsDungeon())
				gain = uint32(gain * 2.75);
			else
				gain *= 2;
		}

		gain = uint32(gain * sWorld->getRate(RATE_XP_KILL));
	}

	sScriptMgr->OnGainCalculation(gain, pl, u);
	return gain;
}

inline float xp_in_group_rate(uint32 count, bool isRaid) {
	float rate;

	if (isRaid) {
		// FIXME: Must apply decrease modifiers depending on raid size.
		rate = 1.0f;
	} else {
		switch (count) {
		case 0:
		case 1:
		case 2:
			rate = 1.0f;
			break;
		case 3:
			rate = 1.166f;
			break;
		case 4:
			rate = 1.3f;
			break;
		case 5:
		default:
			rate = 1.4f;
			break;
		}
	}

	sScriptMgr->OnGroupRateCalculation(rate, count, isRaid);
	return rate;
}
}
}

#endif
